package com.zhf.base;

import com.alibaba.fastjson.JSON;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Map;

/**
 * @Author: ZhaoHuanfeng
 * @Date: 2020/10/26 22:17
 */
public class BaseServlet extends HttpServlet {
    /**
     * doGET方法作为一个调度器，根据请求功能的不同调用对应的方法
     * 规定必须传递一个参数
     * methodName=功能名
     *
     * @param req
     * @param resp
     * @throws ServletException
     * @throws IOException
     */
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

        //获取要访问的方法名
//        String methodName = req.getParameter("methodName");

        String methodName = null;
        //获取Post请求的Conten-Type类型
        String contentType = req.getHeader("Content-Type");

        //判断传入的数据类型
        if ("application/json;charset=utf-8".equalsIgnoreCase(contentType)) {
            //是JSON格式 调用getPostJSON
            String postJSON = getPostJSON(req);

            //将JSON格式的字符串转换为map
            Map<String, Object> map = JSON.parseObject(postJSON, Map.class);

            //从map中获取methodName
            methodName = (String) map.get("methodName");

            //将获取到的数据保存到request域对象中
            req.setAttribute("map", map);
        } else {
            //非JSON的格式
            methodName = req.getParameter("methodName");
        }

        //判断执行对应的方法
//        if ("addCourse".equals(methodName)){
////            addCourse(req, resp);
////        }else if ("findByName".equals(methodName)){
////            findByName(req, resp);
////        }else if ("findByStatus".equals(methodName)){
////            findByStatus(req, resp);
////        }else {
////            System.out.println("请求的功能不存在");
////        }
        if (methodName != null) {
            //通过反射优化代码，提升代码的可维护性
            try {
                //获取字节码对象文件
                Class c = this.getClass();

                //根据传入的方法名获取对应的方法对象
                Method method = c.getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class);

                //调用method对象的invoke方法执行对应功能
                method.invoke(this, req, resp);
            } catch (Exception e) {
                e.printStackTrace();
                System.out.println("请求的功能不存在！");
            }
        }
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        doGet(req, resp);
    }

    /**
     * POST请求格式为：application/json;charset=utf-8
     * 我们使用该方法进行读取
     *
     * @param request
     * @return
     */
    public String getPostJSON(HttpServletRequest request) {
        try {
            //从request中获取缓冲输入流对象
            BufferedReader reader = request.getReader();

            //创建StringBuffer 保存读取出的数据
            StringBuffer stringBuffer = new StringBuffer();

            //循环读取
            String line = null;
            while ((line = reader.readLine()) != null) {
                //将每次读取到的数据追加到Stringbuffer中
                stringBuffer.append(line);
            }

            //返回结果
            return stringBuffer.toString();
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }
}
